package cn.jietuo.first.class03;

/**
 * @author zhangx & jietuo_zx@163.com
 * @version 1.0
 * @date 2020/6/21
 * @description: 给定一个数组arr，和一个整数num。请把小于等于num的数放在数组的左边，大于num的数放在数组的右边。
 * 要求额外空间复杂度O(1)，时间复杂度O(N)
 */
public class Code03_Partition {

    /**
     * 请把小于等于num的数放在数组的左边，大于num的数放在数组的右边。
     *
     * @param arr
     * @param num
     */
    public static void partition(int[] arr, int num) {
        int lessIndex = -1;
        int currIndex = 0;
        while (currIndex < arr.length) {
            if (arr[currIndex] > num) {
                currIndex++;
            } else {
                swap(arr, currIndex++, ++lessIndex);
            }
        }
    }

    /**
     * 请把小于num的数放在数组的左边，大于num的数放在数组的右边，等于num的数放中间
     *
     * @param arr
     * @param num
     */
    public static void partition1(int[] arr, int num) {
        int lessIndex = -1;
        int largeIndex = arr.length;
        int currIndex = 0;
        while (currIndex < largeIndex) {
            if (arr[currIndex] > num) {
                swap(arr, currIndex, largeIndex-- - 1);
            } else if (arr[currIndex] == num) {
                currIndex++;
            } else {
                swap(arr, currIndex++, ++lessIndex);
            }
        }
    }

    private static void swap(int[] arr, int i, int currIndex) {
        int temp = arr[i];
        arr[i] = arr[currIndex];
        arr[currIndex] = temp;
    }

    public static void main(String[] args) {
        int[] arr = new int[]{2, 4, 6, 3, 4, 1, 3};
        printArr(arr);
        partition(arr, 3);
        System.out.println("--------");
        printArr(arr);
    }

    private static void printArr(int[] arr) {
        for (int i = 0; i < arr.length; i++) {
            System.out.println(arr[i]);
        }
    }
}
